package leetcode;

import java.util.HashMap;
import java.util.Map;

/**
 * 437. 路径总和 III
 * 给定一个二叉树的根节点 root ，和一个整数 targetSum ，求该二叉树里节点值之和等于 targetSum 的 路径 的数目。
 * <p>
 * 路径 不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * <p>
 * <p>
 * 输入：root = [10,5,-3,3,2,null,11,3,-2,null,1], targetSum = 8
 * 输出：3
 * 解释：和等于 8 的路径有 3 条，如图所示。
 * 示例 2：
 * <p>
 * 输入：root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
 * 输出：3
 */
public class PathSum2 {


    Map<Integer, Integer> map = new HashMap<>();
    int ans, t;


    // TODO 看不懂
    public int pathSum(TreeNode root, int target) {
        if (root == null) return 0;
        t = target;
        map.put(0, 1);
        dfs(root, root.val);
        return ans;
    }

    void dfs(TreeNode root, int val) {
        // 存在前缀和，value为前缀和出现的次数
        if (map.containsKey(val - t)) {
            ans = ans + map.get(val - t);
        }
        // 到root 一个
        map.put(val, map.getOrDefault(val, 0) + 1);
        if (root.left != null) {
            dfs(root.left, val + root.left.val);
        }
        if (root.right != null) {
            dfs(root.right, val + root.right.val);
        }
        map.put(val, map.getOrDefault(val, 0) - 1);
    }


    /**
     * 迭代法
     * 穷举法，遍历每个节点作为起始节点
     *
     * @param root
     * @param targetSum
     * @return
     */
    public int pathSum1(TreeNode root, int targetSum) {
        if (root == null) {
            return 0;

        }
        int s = rootSum(root, targetSum);
        int l = pathSum1(root.left, targetSum);
        int r = pathSum1(root.right, targetSum);

        return s + l + r;
    }


    /**
     * 求已root为起点有多少种可能
     *
     * @param root
     * @param targetSum
     * @return
     */
    public int rootSum(TreeNode root, int targetSum) {
        int ret = 0;

        if (root == null) {
            return 0;
        }
        int val = root.val;
        if (val == targetSum) {
            ret++;
            // 此处是不能返回，因为后面的节点可能出现负数，即加之后再减
        }

        int left = rootSum(root.left, targetSum - val);
        int right = rootSum(root.right, targetSum - val);
        return ret + left + right;
    }


    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;
    }
}
